package com.atguigu.gmall.realtime.app;

import com.atguigu.gmall.realtime.util.FlinkSourceUtil;
import org.apache.flink.configuration.Configuration;
import org.apache.flink.runtime.state.hashmap.HashMapStateBackend;
import org.apache.flink.streaming.api.CheckpointingMode;
import org.apache.flink.streaming.api.datastream.DataStreamSource;
import org.apache.flink.streaming.api.environment.CheckpointConfig;
import org.apache.flink.streaming.api.environment.StreamExecutionEnvironment;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;

/*
 * @title: BaseAppV2
 * @projectName gmal0924
 * @description: TODO
 * @author liuyun
 * @date 2022-3-21 10:33
 */
abstract public class BaseAppV2 {

    public void init(int port, int p, String ck, String groupId, String fistToPic, String... otherTopics) {
        System.setProperty("HADOOP_USER_NAME", "atguigu");
        Configuration conf = new Configuration();
        conf.setInteger("rest.port", port);
        StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment(conf);
        env.setParallelism(p);

        // 开启checkpoint
        env.enableCheckpointing(3000);
        // 设置状态后端
        env.setStateBackend(new HashMapStateBackend());
        env.getCheckpointConfig().setCheckpointStorage("hdfs://hadoop162:8020/ck/gmall/" + ck);

        env.getCheckpointConfig().setMaxConcurrentCheckpoints(1);
        env.getCheckpointConfig().setMinPauseBetweenCheckpoints(500);
        env.getCheckpointConfig().setCheckpointingMode(CheckpointingMode.EXACTLY_ONCE);
        env.getCheckpointConfig().setCheckpointTimeout(30 * 1000);
        env.getCheckpointConfig().setExternalizedCheckpointCleanup(
                CheckpointConfig.ExternalizedCheckpointCleanup.RETAIN_ON_CANCELLATION);


        ArrayList<String> topics = new ArrayList<>();
        topics.add(fistToPic);
        topics.addAll(Arrays.asList(otherTopics));

        HashMap<String, DataStreamSource<String>> topicStreamMap = new HashMap<>();

        for (String topic : topics) {
            DataStreamSource<String> stream = env.addSource(FlinkSourceUtil.getKafkaSource(groupId, topic));
            topicStreamMap.put(topic, stream);
        }

        handle(env, topicStreamMap);

        try {
            env.execute(ck);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    protected abstract void handle(StreamExecutionEnvironment env, HashMap<String, DataStreamSource<String>> topicStreamMap);
}
